import torch
from CustomImageDataset import CustomImageDataset
from torch.utils.data import Dataset

dataset = CustomImageDataset('data/label.csv', 'data/images')
